/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * 
 */

package org.apache.jmeter.visualizers;

import java.io.Serializable;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Vector;

import org.apache.jmeter.samplers.SampleResult;
import org.apache.jorphan.logging.LoggingManager;
import org.apache.jorphan.math.StatCalculator;
import org.apache.log.Logger;

/**
 * Aggegate sample data container. Just instantiate a new instance of this
 * class, and then call {@link #addSample(SampleResult)} a few times, and pull
 * the stats out with whatever methods you prefer.
 * 
 */
public class SamplingStatCalculator implements Serializable {
	private static final Logger log = LoggingManager.getLoggerForClass();

	private final StatCalculator calculator = new StatCalculator();

	private final List storedValues = new Vector();

	private double maxThroughput;

	private long firstTime;

	private String label;

	public SamplingStatCalculator() {// Don't (can't) use this...
        log.warn("Constructor only intended for use in testing"); // $NON-NLS-1$
	}

	/**
	 * Use this constructor.
	 */
	public SamplingStatCalculator(String label) {
		this.label = label;
		init();
	}

	/**
	 * Essentially a copy function
	 * 
	 * @param stat
	 */
	public SamplingStatCalculator(SamplingStatCalculator stat) {
		this(stat.label);
		addSamples(stat);
	}

	private void init() {
		firstTime = Long.MAX_VALUE;
		calculator.clear();
		storedValues.clear();
		maxThroughput = Double.MIN_VALUE;
	}

	public void addSamples(SamplingStatCalculator ssc) {
		calculator.addAll(ssc.calculator);
        synchronized( storedValues )
        {
            storedValues.addAll(ssc.storedValues);
            Collections.sort(storedValues);
        }
        if (firstTime > ssc.firstTime) {
			firstTime = ssc.firstTime;
		}
	}

	/**
	 * Clear the counters (useful for differential stats)
	 * 
	 */
	public synchronized void clear() {
		init();
	}

	public Sample getCurrentSample() {
        synchronized( storedValues )
        {
            if (storedValues.size() == 0) {
                return new Sample();
            }
            return (Sample) storedValues.get(storedValues.size() - 1);
        }
	}

	/**
	 * Get the elapsed time for the samples
	 * 
	 * @return how long the samples took
	 */
	public long getElapsed() {
		if (getCurrentSample().getEndTime() == 0) {
			return 0;// No samples collected ...
		}
		return getCurrentSample().getEndTime() - firstTime;
	}

	/**
	 * Returns the throughput associated to this sampler in requests per second.
	 * May be slightly skewed because it takes the timestamps of the first and
	 * last samples as the total time passed, and the test may actually have
	 * started before that start time and ended after that end time.
	 */
	public double getRate() {
		if (calculator.getCount() == 0) {
			return 0.0; // Better behaviour when howLong=0 or lastTime=0
		}

		return getCurrentSample().getThroughput();
	}

	/**
	 * Should calculate the average page size, which means divide the bytes by number
	 * of samples - actually calculates the throughput in bytes / second
	 * 
     * @deprecated use getBytesPerSecond() instead
     * @see #getAvgPageBytes() for the average page size
	 */
	public double getPageSize() {
		double rate = 0;
		if (this.getElapsed() > 0 && calculator.getTotalBytes() > 0) {
			rate = calculator.getTotalBytes() / ((double) this.getElapsed() / 1000);
		}
		if (rate < 0) {
			rate = 0;
		}
		return rate;
	}

    /**
     * Throughput in bytes / second
     * 
     * @return throughput in bytes/second
     */
    public double getBytesPerSecond() {
    	// Code duplicated from getPageSize()
		double rate = 0;
		if (this.getElapsed() > 0 && calculator.getTotalBytes() > 0) {
			rate = calculator.getTotalBytes() / ((double) this.getElapsed() / 1000);
		}
		if (rate < 0) {
			rate = 0;
		}
		return rate;
    }

    /**
     * Throughput in kilobytes / second
     * 
     * @return Throughput in kilobytes / second
     */
    public double getKBPerSecond() {
        return getBytesPerSecond() / 1024; // 1024=bytes per kb
    }

    /**
     * calculates the average page size, which means divide the bytes by number
     * of samples.
     * 
     * @return average page size in bytes
     */
	public double getAvgPageBytes() {
		double rate = 0;
		if (this.getElapsed() > 0 && calculator.getTotalBytes() > 0) {
			rate = calculator.getTotalBytes() / ((double) this.getElapsed() / 1000);
		}
		if (rate < 0) {
			rate = 0;
		}
		return rate;
	}

	public String getLabel() {
		return label;
	}

	/**
	 * Records a sample.
	 * 
	 */
	public Sample addSample(SampleResult res) {
        long rtime, cmean, cstdv, cmedian, cpercent, eCount, endTime;
        double throughput;
        boolean rbool;
        synchronized (calculator) {
			long byteslength = res.getBytes();
			// if there was more than 1 loop in the sample, we
			// handle it appropriately
			if (res.getSampleCount() > 1) {
				long time = res.getTime() / res.getSampleCount();
				long resbytes = byteslength / res.getSampleCount();
				for (int idx = 0; idx < res.getSampleCount(); idx++) {
					calculator.addValue(time);
					calculator.addBytes(resbytes);
				}
			} else {
				calculator.addValue(res.getTime());
				calculator.addBytes(byteslength);
			}
			setStartTime(res);
			eCount = getCurrentSample().getErrorCount();
			if (!res.isSuccessful()) {
				eCount++;
			}
			endTime = getEndTime(res);
			long howLongRunning = endTime - firstTime;
			throughput = ((double) calculator.getCount() / (double) howLongRunning) * 1000.0;
			if (throughput > maxThroughput) {
				maxThroughput = throughput;
			}

            rtime = res.getTime();
            cmean = (long)calculator.getMean();
            cstdv = (long)calculator.getStandardDeviation();
            cmedian = calculator.getMedian().longValue();
            cpercent = calculator.getPercentPoint( 0.500 ).longValue();
// TODO cpercent is the same as cmedian here - why? and why pass it to "distributionLine"? 
            rbool = res.isSuccessful();
        }

        synchronized( storedValues ){
            int count = storedValues.size() + 1;
            Sample s =
                new Sample( null, rtime, cmean, cstdv, cmedian, cpercent, throughput, eCount, rbool, count, endTime );
            storedValues.add( s );
            return s;
        }
	}

	public List getSamples() {
		return storedValues;
	}

	public Sample getSample(int index) {
        synchronized( storedValues ){
            if (index < storedValues.size()) {
                return (Sample) storedValues.get(index);
            }
		return null;
        }
	}

	private long getEndTime(SampleResult res) {
		long endTime = res.getEndTime();
		long lastTime = getCurrentSample().getEndTime();
		if (lastTime < endTime) {
			lastTime = endTime;
		}
		return lastTime;
	}

	/**
	 * @param res
	 */
	private void setStartTime(SampleResult res) {
		long startTime = res.getStartTime();
		if (firstTime > startTime) {
			// this is our first sample, set the start time to current timestamp
			firstTime = startTime;
		}
	}

	/**
	 * Returns the raw double value of the percentage of samples with errors
	 * that were recorded. (Between 0.0 and 1.0)
	 * 
	 * @return the raw double value of the percentage of samples with errors
	 *         that were recorded.
	 */
	public double getErrorPercentage() {
		double rval = 0.0;

		if (calculator.getCount() == 0) {
			return (rval);
		}
		rval = (double) getCurrentSample().getErrorCount() / (double) calculator.getCount();
		return (rval);
	}

	/**
	 * For debugging purposes, only.
	 */
	public String toString() {
		StringBuffer mySB = new StringBuffer();

		mySB.append("Samples: " + this.getCount() + "  ");
		mySB.append("Avg: " + this.getMean() + "  ");
		mySB.append("Min: " + this.getMin() + "  ");
		mySB.append("Max: " + this.getMax() + "  ");
		mySB.append("Error Rate: " + this.getErrorPercentage() + "  ");
		mySB.append("Sample Rate: " + this.getRate());
		return (mySB.toString());
	}

	/**
	 * @return errorCount
	 */
	public long getErrorCount() {
		return getCurrentSample().getErrorCount();
	}

	/**
	 * @return Returns the maxThroughput.
	 */
	public double getMaxThroughput() {
		return maxThroughput;
	}

	public HashMap getDistribution() {
		return calculator.getDistribution();
	}

	public Number getPercentPoint(double percent) {
		return calculator.getPercentPoint(percent);
	}

	public int getCount() {
		return calculator.getCount();
	}

	public Number getMax() {
		return calculator.getMax();
	}

	public double getMean() {
		return calculator.getMean();
	}

	public Number getMeanAsNumber() {
		return new Long((long) calculator.getMean());
	}

	public Number getMedian() {
		return calculator.getMedian();
	}

	public Number getMin() {
		if (calculator.getMin().longValue() < 0) {
			return new Long(0);
		}
		return calculator.getMin();
	}

	public Number getPercentPoint(float percent) {
		return calculator.getPercentPoint(percent);
	}

	public double getStandardDeviation() {
		return calculator.getStandardDeviation();
	}
}
